package tree;

import java.util.ArrayList;
import java.util.List;

/**
 * @author: june
 * @date: 2023/7/18
 * <p>
 * 501. 二叉搜索树中的众数
 * <p>
 * 给你一个含重复值的二叉搜索树（BST）的根节点 root ，找出并返回 BST 中的所有 众数（即，出现频率最高的元素）。
 * <p>
 * 如果树中有不止一个众数，可以按 任意顺序 返回。
 * <p>
 * 假定 BST 满足如下定义：
 * <p>
 * 结点左子树中所含节点的值 小于等于 当前节点的值
 * 结点右子树中所含节点的值 大于等于 当前节点的值
 * 左子树和右子树都是二叉搜索树
 **/

public class FindModeInBst_501 {

    TreeNode pre;
    int maxCnt = 0;
    int tmpCnt = 0;
    List<Integer> list = new ArrayList<>();

    public int[] findMode(TreeNode root) {
        traversal(root);
        int[] result = new int[list.size()];
        for (int i = 0; i < list.size(); i++) {
            result[i] = list.get(i);
        }
        return result;
    }

    // 二叉搜索树中序遍历后得到有序数组，一次遍历得到有序数组的众数
    private void traversal(TreeNode root) {
        if (root == null) {
            return;
        }

        traversal(root.left);
        if (pre == null || pre.val != root.val) {
            tmpCnt = 1;
        }

        if (pre != null && pre.val == root.val) {
            tmpCnt++;
        }

        if (tmpCnt == maxCnt) {
            list.add(root.val);
        } else if (tmpCnt > maxCnt) {
            list.clear();
            list.add(root.val);
            maxCnt = tmpCnt;
        }

        pre = root;
        traversal(root.right);
    }
}
